#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
AI营销方案生成系统 - 路由模块
功能：路由注册和管理模块的包初始化文件
作者：AI营销系统开发团队
创建时间：2024
"""

from flask import Flask
from flask_socketio import SocketIO

from .template_routes import template_bp
from .requirement_routes import requirement_bp
from .chat_routes import chat_bp
from .websocket_handlers import register_websocket_handlers


def register_routes(app: Flask, socketio: SocketIO) -> None:
    """
    注册所有路由和WebSocket处理器
    
    Args:
        app: Flask应用实例
        socketio: SocketIO实例
    """
    # 注册蓝图路由
    app.register_blueprint(template_bp, url_prefix='/api')
    app.register_blueprint(requirement_bp, url_prefix='/api')
    app.register_blueprint(chat_bp, url_prefix='/api')
    
    # 注册WebSocket处理器
    register_websocket_handlers(socketio)
    
    # 注册根路由
    @app.route('/')
    def index():
        return {
            'message': 'AI营销方案生成系统API服务',
            'version': '1.0.0',
            'status': 'running'
        }
    
    # 健康检查路由
    @app.route('/health')
    def health_check():
        return {
            'status': 'healthy',
            'service': 'ai-marketiq-backend'
        }


__all__ = ['register_routes']